package com.imooc.files.controller;

import com.imooc.api.controller.files.FileUploaderControllerApi;
import com.imooc.exception.GraceException;
import com.imooc.files.resource.FileResource;
import com.imooc.files.service.UploaderService;
import com.imooc.grace.result.GraceJSONResult;
import com.imooc.grace.result.ResponseStatusEnum;
import com.imooc.pojo.bo.NewAdminBO;
import com.imooc.utils.FileUtils;
import com.mongodb.client.gridfs.GridFSBucket;
import com.mongodb.client.gridfs.GridFSFindIterable;
import com.mongodb.client.gridfs.model.GridFSFile;
import com.mongodb.client.model.Filters;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.lang3.StringUtils;
import org.bson.types.ObjectId;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.util.ObjectUtils;
import org.springframework.web.bind.annotation.RestController;
import org.springframework.web.multipart.MultipartFile;
import sun.misc.BASE64Decoder;

import javax.servlet.http.HttpServletResponse;
import java.io.ByteArrayInputStream;
import java.io.File;
import java.io.FileOutputStream;
import java.io.OutputStream;
import java.util.ArrayList;
import java.util.List;

@Slf4j
@RestController
public class FileUploaderController implements FileUploaderControllerApi {

    @Autowired
    private FileResource fileResource;

    @Autowired
    private GridFSBucket gridFSBucket;

    @Autowired
    private UploaderService uploaderService;

    /**
     * 上传单文件
     */
    @Override
    public GraceJSONResult uploadFace(String userId, MultipartFile file) throws Exception {
        // 文件流不能为空
        if (ObjectUtils.isEmpty(file)) {
            return GraceJSONResult.errorCustom(ResponseStatusEnum.FILE_UPLOAD_NULL_ERROR);
        }

        // 获得文件上传的名称, 并且不能为空
        String fileName = file.getOriginalFilename();
        if (StringUtils.isBlank(fileName)) {
            return GraceJSONResult.errorCustom(ResponseStatusEnum.FILE_UPLOAD_NULL_ERROR);
        }

        // 将文件名切割成数组
        String[] fileNameArr = fileName.split("\\.");
        // 获得文件的后缀, 例如: jpg、png...
        String suffix = fileNameArr[fileNameArr.length - 1];
        // 判断后缀符合我们的预定义规范, 支持 png、jpg、jpeg
        if (!suffix.equalsIgnoreCase("png") && !suffix.equalsIgnoreCase("jpg") && !suffix.equalsIgnoreCase("jpeg")) {
            return GraceJSONResult.errorCustom(ResponseStatusEnum.FILE_FORMATTER_FAILD);
        }

        // fixme 文件上传前可以需要对, 文件进行校验, 可以使用阿里的校验服务, 都是调包侠这里就不做了

        // 执行文件上传, 底层使用什么存储方式无所谓, 上述的处理逻辑是基于自己的业务
        String path = this.uploaderService.uploadOSS(file, userId, suffix);
        log.info("FileUploaderController#uploadFace: The path returned by the file upload, path = {}", path);
        if (StringUtils.isBlank(path)) {
            return GraceJSONResult.errorCustom(ResponseStatusEnum.FILE_UPLOAD_FAILD);
        }

        // 返回组装好的路径
        String finalPath = fileResource.getHost() + path;
        return GraceJSONResult.ok(finalPath);
    }

    /**
     * 上传多个文件
     */
    @Override
    public GraceJSONResult uploadSomeFiles(String userId, MultipartFile[] files) throws Exception {
        // 文件流不能为空
        if (ObjectUtils.isEmpty(files) || files.length <= 0) {
            return GraceJSONResult.errorCustom(ResponseStatusEnum.FILE_UPLOAD_NULL_ERROR);
        }

        // 声明list，用于存放多个图片的地址路径，返回到前端
        List<String> imageUrlList = new ArrayList<>();

        // 内部和上传单张没啥区别, 这里就不重复写了
        for (MultipartFile file : files) {
            GraceJSONResult graceJSONResult = this.uploadFace(userId, file);
            imageUrlList.add((String) graceJSONResult.getData());
        }

        // 返回文件地址
        return GraceJSONResult.ok(imageUrlList);
    }

    /**
     * 文件上传到MongoDB的GriDFS中
     * 文件上传到MongoDB的GridFS中
     */
    @Override
    public GraceJSONResult uploadToGridFS(NewAdminBO newAdminBO) throws Exception {
        // 获得图片的base64字符串
        String file64 = newAdminBO.getImg64();
        // 将base64字符串转换为byte数组
        byte[] bytes = new BASE64Decoder().decodeBuffer(file64.trim());
        // 转换为输入流
        ByteArrayInputStream inputStream = new ByteArrayInputStream(bytes);
        // 上传到GriDFS中, 参数一: 文件名称、参数二: 文件流
        ObjectId fileId = gridFSBucket.uploadFromStream(newAdminBO.getUsername() + ".png", inputStream);
        // 获得文件在GriDFS中的主键id
        String fileIdStr = fileId.toString();
        // 返回文件id
        return GraceJSONResult.ok(fileIdStr);
    }

    /**
     * 从GriDFS中读取图片内容
     * 从GridFS中读取图片内容, 对应的业务是查看人脸数据
     */
    @Override
    public void readInGridFS(String faceId, HttpServletResponse response) throws Exception {
        // 判断参数
        if (StringUtils.isBlank(faceId) || faceId.equalsIgnoreCase("null")) {
            GraceException.display(ResponseStatusEnum.FILE_NOT_EXIST_ERROR);
        }

        // 从GriDFS中读取
        File adminFace = readGridFSByFaceId(faceId);

        // 把人脸图片输出到浏览器
        FileUtils.downloadFileByStream(response, adminFace);
    }

    /**
     * 从GriFS中读取图片内容，并且返回base64
     */
    @Override
    public GraceJSONResult readFace64InGridFS(String faceId) throws Exception {
        // 获得GriDFS中人脸文件
        File myFace = readGridFSByFaceId(faceId);
        // 转换人脸为base64
        String base64Face = FileUtils.fileToBase64(myFace);
        return GraceJSONResult.ok(base64Face);
    }

    /**
     * 读取GriFS中的文件数据
     */
    /**
     * 从GridFS中读取数据
     * @param faceId
     * @return
     * @throws Exception
     */
    private File readGridFSByFaceId(String faceId) throws Exception {
        // 相当于查找 "_id" = faceId 的数据
        GridFSFindIterable gridFSFiles = gridFSBucket.find(Filters.eq("_id", new ObjectId(faceId)));
        GridFSFile gridFS = gridFSFiles.first();
        if (gridFS == null) {
            GraceException.display(ResponseStatusEnum.FILE_NOT_EXIST_ERROR);
        }

        String fileName = gridFS.getFilename();

        // 获取文件流，保存文件到本地或者服务器的临时目录
        File fileTemp = new File("/data/temp_face");
        if (!fileTemp.exists()) {
            boolean mkdirs = fileTemp.mkdirs();
        }

        File myFile = new File("/data/temp_face/" + fileName);

        // 创建文件输出流
        OutputStream os = new FileOutputStream(myFile);

        // 下载到服务器或者本地
        gridFSBucket.downloadToStream(new ObjectId(faceId), os);
        return myFile;
    }
}
